import lightgbm as lgb

def get_lgbm_model(n_estimators=100, random_state=0):
    return lgb.LGBMClassifier(
n_estimators=n_estimators,
random_state = random_state,
importance_type = 'gain',
verbose = -1  # 关闭训练日志
)